package com.example.demo.mqserver.core;

import com.example.demo.common.ConsumerEnv;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.Data;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * Created with IntelliJ IDEA.
 * Description:
 * User: 张
 * Date: 2024-07-10
 * Time: 19:29
 */
//存储消息队列
@Data
public class MSGQueue {
    //表示队列的身份标识
    private String name;
    //表示队列是否持久化  ture表示持久化保存, false表示不持久化
    private boolean durable = false;
    //这个属性为ture 表示这个队列只能被一个消费者使用
    //先不实现
    private boolean exclusive = false;
    private boolean autoDelete = false;
    private Map<String,Object> arguments = new HashMap<>();

    //扩展属性来表示消费者(当前队列都有哪些消费者订阅)
    private List<ConsumerEnv> consumerEnvList = new ArrayList<>();

    //记录当前取到第几个消费者
    //AtomicInteger是int类型的原子化操作,因为消息的消费存在多线陈调用的问题(线程安全),所以使用AtomicInteger 而不是 int
    private AtomicInteger consumerSeq = new AtomicInteger(0);

    //添加新的订阅者  后续使用这个方法的时候,已经针对queue进行加锁了,此处就不进行上锁
    public void addConsumerEnv(ConsumerEnv consumerEnv) {
        consumerEnvList.add(consumerEnv);
    }

    //挑选一个订阅者,用来处理当前的消息(按照轮询的方式来进行挑选)
    //这个方法只是单纯的针对list进行读操作,不是写,此处不上锁也是可以的
    //consumerSeq 一开始就是原子性的,更不用上锁了
    public ConsumerEnv choseConsumer() {
        if (consumerEnvList.size() == 0) {
            return null; //该队列没人订阅
        }
        //计算当前要去的元素下标
        int index = consumerSeq.get() % consumerEnvList.size();
        consumerSeq.getAndIncrement();//自增(相当于++)
        return consumerEnvList.get(index);
    }


    //数据库专用setter getter
    //arguments序列化
    public String getArguments() {
        //把当前的argument通过json转换成String
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        return "{}";//如果代码出现异常
    }

    //从数据库读数据的之后,构造Exchange对象
    public void setArguments(String argumentsJson) {
        //把参数中的argumentsJson按照json格式解析
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            this.arguments = objectMapper.readValue(argumentsJson, new TypeReference<HashMap<String,Object>>() {});//通过TypeReference这个方法来描述当前的的数据要转换成啥样的对象
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
    }

    // java 内部使用set get
    public Object getArguments(String key) {
        return arguments.get(key);
    }

    public void setArguments(String key,Object value) {
        arguments.put(key,value);
    }

    public void setArguments(Map<String,Object> arguments) {
        this.arguments = arguments;
    }
}
